# -*- coding: utf-8 -*-
# author: liangzilong
from flask import g

"""
自定义一个装饰器，判断用户是否登录了。如果登录，继续访问。如果没有登录直接返回（不能继续访问）
本质上就是一个登录的拦截器
"""


def login_required(func):
    def wrapper(*args, **kwargs):

        if g.user_id is not None:  # 表示用户已经登录过
            return func(*args, **kwargs)
        else:
            return {'message': '用户没有登录，不能继续访问'}, 401

    return wrapper
